from config import *
import pymysql

# 创建数据库的连接
# host:我们要连接的主机
conn = pymysql.connect(host=host,
                       user=user,
                       passwd=passwd,
                       port=3306,
                       db=db
                       )

# 创建游标   =>  真正执行sql语句
# 开启了一个隐形的事务
cur = conn.cursor()

# -------------------------------------------创建表--------------------------------------------
# IP流量表（总表）
# 先删除同名表
sql = "DROP TABLE IF EXISTS ip_info"
cur.execute(sql)   # 执行sql语句
# 创建表 如果存在就不创建， 起初会删除同名的数据表（初始化）
sql = """
CREATE TABLE ip_info(
    ip_address VARCHAR(255),
    pro VARCHAR(255),
    Operator VARCHAR(255),
    used_date VARCHAR(255),
    flow_rate FLOAT
)
"""
cur.execute(sql)

# 省份表（子表）
# 先删除同名表
sql = "DROP TABLE IF EXISTS pro"
cur.execute(sql)   # 执行sql语句
# 创建表 如果存在就不创建， 起初会删除同名的数据表（初始化）
sql = """
CREATE TABLE pro(
    id INT AUTO_INCREMENT PRIMARY KEY,
    pro_name VARCHAR(255)
)
"""
cur.execute(sql)
# 插入省份
sql = """
INSERT INTO pro(pro_name) VALUES (%s)
"""
cur.executemany(sql, provinces)

# 新建运营商表
# 先删除同名表
sql = "DROP TABLE IF EXISTS operator"
cur.execute(sql)   # 执行sql语句
# 创建表 如果存在就不创建， 起初会删除同名的数据表（初始化）
sql = """
CREATE TABLE operator(
    id INT AUTO_INCREMENT PRIMARY KEY,
    operator_name VARCHAR(255)
)
"""
cur.execute(sql)

# 插入数据
sql = """
INSERT INTO operator(operator_name) VALUES (%s)
"""
cur.executemany(sql, operators)

conn.commit()
conn.close()